package com.zxlfysj.horse;

import java.awt.*;
import java.util.ArrayList;
import java.util.Comparator;

/**
 * 马踏棋盘
 *
 * @author yangshujing
 * @create 2020-08-26 22:38
 */
public class HorseChessboard {
    private static int X;   //棋盘的列数
    private static int Y;   //棋盘的行数
    private static boolean[] visited; //标记棋盘的位置是否被访问过
    private static boolean finished; //如果是true，成功

    public static void main(String[] args) {
        //测试
        X = 8;
        Y = 8;
        int row = 2;
        int column = 4;
        int[][] chessboard = new int[X][Y];
        visited = new boolean[X * Y];
        traversalChessboard(chessboard, row - 1, column - 1, 1);

        //输出棋盘的情况
        for(int[] rows : chessboard) {
            for(int step : rows) {
                System.out.print(step + "\t");
            }
            System.out.println();
        }

    }

    /**
     * 马踏棋盘算法
     *
     * @param chessboard 棋盘，二维数组表示
     * @param row        马儿当期的位置的行 从0开始
     * @param column     马儿当期的位置的列 从0开始
     * @param step       走到第几步了，初始位置是第一步
     */
    public static void traversalChessboard(int[][] chessboard, int row, int column, int step) {
        chessboard[row][column] = step;
        visited[row * X + column] = true;
        //获取当期位置可以走的位置的集合
        ArrayList<Point> ps = next(new Point(column, row));
        //对ps进行排序，就是先选择走ps的所有的元素的下一步的位置个数最少的，这样减少回溯次数（贪心算法）
        sort(ps);
        while (!ps.isEmpty()) {
            Point p = ps.remove(0); //取出下一个可以走的位置
            if (!visited[p.y * X + p.x]) {
                traversalChessboard(chessboard, p.y, p.x, step + 1);
            }
        }
        //没走完或者需要回溯
        if (step < X * Y && !finished) {
            chessboard[row][column] = 0;
            visited[row * X + column] = false;
        } else {
            finished = true;
        }

    }

    /**
     * 在当前位置还能走那些位置，最多有8个位置
     *
     * @param curPoint
     * @return
     */
    public static ArrayList<Point> next(Point curPoint) {
        ArrayList<Point> ps = new ArrayList<>();
        Point p1 = new Point();
        //判断马是否可以走日的位置，可以就加入集合中
        if ((p1.x = curPoint.x - 2) >= 0 && (p1.y = curPoint.y - 1) >= 0) {
            ps.add(new Point(p1));
        }
        if ((p1.x = curPoint.x - 1) >= 0 && (p1.y = curPoint.y - 2) >= 0) {
            ps.add(new Point(p1));
        }
        if ((p1.x = curPoint.x + 1) < X && (p1.y = curPoint.y - 2) >= 0) {
            ps.add(new Point(p1));
        }
        if ((p1.x = curPoint.x + 2) < X && (p1.y = curPoint.y - 1) >= 0) {
            ps.add(new Point(p1));
        }
        if ((p1.x = curPoint.x + 2) < X && (p1.y = curPoint.y + 1) < Y) {
            ps.add(new Point(p1));
        }
        if ((p1.x = curPoint.x + 1) < X && (p1.y = curPoint.y + 2) < Y) {
            ps.add(new Point(p1));
        }
        if ((p1.x = curPoint.x - 1) >= 0 && (p1.y = curPoint.y + 2) < Y) {
            ps.add(new Point(p1));
        }
        if ((p1.x = curPoint.x - 2) >= 0 && (p1.y = curPoint.y + 1) < Y) {
            ps.add(new Point(p1));
        }
        return ps;
    }

    //根据当前这一步的下一步的选择位置，进行非递减排序
    public static void sort(ArrayList<Point> ps) {
        ps.sort(new Comparator<Point>() {
            @Override
            public int compare(Point o1, Point o2) {
                //获取o1的下一步的所有可以走的位置的个数
                int count1 = next(o1).size();
                int count2 = next(o2).size();
                if(count1 < count2) {
                    return -1;
                } else if(count1 == count2) {
                    return 0;
                } else {
                    return 1;
                }
            }
        });
    }
}
